function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }

function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }

function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }

function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }

function _iterableToArrayLimit(arr, i) { var _i = arr == null ? null : typeof Symbol !== "undefined" && arr[Symbol.iterator] || arr["@@iterator"]; if (_i == null) return; var _arr = []; var _n = true; var _d = false; var _s, _e; try { for (_i = _i.call(arr); !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; }

function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }

function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); if (enumerableOnly) { symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; }); } keys.push.apply(keys, symbols); } return keys; }

function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i] != null ? arguments[i] : {}; if (i % 2) { ownKeys(Object(source), true).forEach(function (key) { _defineProperty(target, key, source[key]); }); } else if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } } return target; }

function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }

function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }

function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }

import { Modal, message, Input, Upload } from 'antd';
import { DataUri } from '@antv/x6';
import { nextTick, getGraphJSONData, autoSaveJson, launchFullScreen, exitFullscreen } from './tools';
import React from 'react';
const Dragger = Upload.Dragger; // 文件解析

const fileExplain = file => {
  const reader = new FileReader();
  reader.readAsText(file);

  reader.onload = /*#__PURE__*/function () {
    var _ref = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee(event) {
      var _event$target, jsonData;

      return regeneratorRuntime.wrap(function _callee$(_context) {
        while (1) switch (_context.prev = _context.next) {
          case 0:
            Modal.destroyAll();
            _context.prev = 1;
            jsonData = JSON.parse(((_event$target = event.target) === null || _event$target === void 0 ? void 0 : _event$target.result) || '[]');
            message.success('文件解析完成');
            _context.next = 6;
            return autoSaveJson(jsonData, null);

          case 6:
            window.location.reload();
            _context.next = 12;
            break;

          case 9:
            _context.prev = 9;
            _context.t0 = _context["catch"](1);
            message.error('文件解析失败');

          case 12:
          case "end":
            return _context.stop();
        }
      }, _callee, null, [[1, 9]]);
    }));

    return function (_x) {
      return _ref.apply(this, arguments);
    };
  }();

  Modal.info({
    closable: false,
    className: 'rasir-fc-model',
    title: '文件解析',
    content: '文件解析中，请稍等...',
    okButtonProps: {
      disabled: true
    }
  });
};

export default function useHooks({
  onRemoveLink,
  onRemoveNode,
  // onAddNode,
  onAddLink,
  onSave
}) {
  const actionFunMap = {
    // 编辑节点
    edite: (_, __, config) => {
      const setVisible = config.setVisible;
      setVisible(true);
    },
    // 删除节点或连线
    delete: (cell, graph) => {
      let cellItems = [];
      if (cell) cellItems = [cell];else {
        // 如果currentCell没有值 就去graph中找有没有被选中的节点
        cellItems = graph.getSelectedCells();
      }

      if (!cellItems.length) {
        message.info('请先选择要删除的节点', {
          duration: 1000
        });
        return;
      }

      const isNode = cellItems[0].isNode();
      const isEdge = cellItems[0].isEdge();

      if (isNode) {
        Modal.confirm({
          title: '删除节点',
          className: 'rasir-fc-model',
          content: `该操作将删除节点 ${cellItems.map(cell => cell.label).join(',')} 是否继续？`,
          okText: '确定',
          cancelText: '取消',
          centered: true,
          onOk: () => new Promise( /*#__PURE__*/function () {
            var _ref2 = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee2(resolve, reject) {
              return regeneratorRuntime.wrap(function _callee2$(_context2) {
                while (1) switch (_context2.prev = _context2.next) {
                  case 0:
                    _context2.t0 = !onRemoveNode;

                    if (_context2.t0) {
                      _context2.next = 8;
                      break;
                    }

                    _context2.t1 = onRemoveNode;

                    if (!_context2.t1) {
                      _context2.next = 7;
                      break;
                    }

                    _context2.next = 6;
                    return onRemoveNode(cellItems.map(cell => cell.data));

                  case 6:
                    _context2.t1 = _context2.sent;

                  case 7:
                    _context2.t0 = _context2.t1;

                  case 8:
                    if (!_context2.t0) {
                      _context2.next = 13;
                      break;
                    }

                    graph.removeCells(cellItems.map(cell => graph.getCellById(cell.data.id)));
                    resolve(1);
                    _context2.next = 14;
                    break;

                  case 13:
                    reject();

                  case 14:
                  case "end":
                    return _context2.stop();
                }
              }, _callee2);
            }));

            return function (_x2, _x3) {
              return _ref2.apply(this, arguments);
            };
          }())
        });
      } else if (isEdge) {
        Modal.confirm({
          title: '删除连线',
          className: 'rasir-fc-model',
          content: `该操作将删除连线是否继续？请确认`,
          okText: '确定',
          cancelText: '取消',
          centered: true,
          onOk: () => new Promise( /*#__PURE__*/function () {
            var _ref3 = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee3(resolve, reject) {
              return regeneratorRuntime.wrap(function _callee3$(_context3) {
                while (1) switch (_context3.prev = _context3.next) {
                  case 0:
                    _context3.t0 = onRemoveLink;

                    if (!_context3.t0) {
                      _context3.next = 5;
                      break;
                    }

                    _context3.next = 4;
                    return onRemoveLink(cellItems);

                  case 4:
                    _context3.t0 = !_context3.sent;

                  case 5:
                    if (!_context3.t0) {
                      _context3.next = 9;
                      break;
                    }

                    reject();
                    _context3.next = 11;
                    break;

                  case 9:
                    graph.removeCells(cellItems);
                    resolve(1);

                  case 11:
                  case "end":
                    return _context3.stop();
                }
              }, _callee3);
            }));

            return function (_x4, _x5) {
              return _ref3.apply(this, arguments);
            };
          }()),

          onCancel() {
            cellItems.forEach(cellItem => {
              graph.addCell(cellItem);
            });
          }

        });
      }
    },
    // 添加节点或连线
    add: function () {
      var _add = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee4(cellItem, graph) {
        var isEdge, _ref4, source, target, edgeData;

        return regeneratorRuntime.wrap(function _callee4$(_context4) {
          while (1) switch (_context4.prev = _context4.next) {
            case 0:
              isEdge = cellItem.isEdge(); // 暂时没用上 后续可优化
              // const isNode = cellItem.isNode();
              // if (isNode) {
              //   message.loading({ duration: 0, content: '节点生成中，请稍等' });
              //   let ret: any;
              //   if (onAddNode) ret = await onAddNode();
              //   message.destroy();
              //   if (onAddNode && !ret) {
              //     graph.removeCell(cellItem);
              //   } else message.success('节点生成成功');
              // } else

              if (!isEdge) {
                _context4.next = 14;
                break;
              }

              message.loading({
                duration: 0,
                content: '连接生成中，请稍等'
              });
              _context4.t0 = onAddLink;

              if (!_context4.t0) {
                _context4.next = 8;
                break;
              }

              _context4.next = 7;
              return onAddLink();

            case 7:
              _context4.t0 = !_context4.sent;

            case 8:
              if (_context4.t0) {
                _context4.next = 14;
                break;
              }

              _ref4 = cellItem, source = _ref4.source, target = _ref4.target;
              edgeData = {
                id: `${source.cell}_${target.cell}`,
                source: source.cell,
                target: target.cell
              };
              graph.addEdge(_objectSpread({
                shape: 'lane-edge',
                data: _objectSpread(_objectSpread({}, edgeData), {}, {
                  type: 'link',
                  zIndex: 1
                })
              }, edgeData));
              message.destroy();
              message.success('连接成功');

            case 14:
            case "end":
              return _context4.stop();
          }
        }, _callee4);
      }));

      function add(_x6, _x7) {
        return _add.apply(this, arguments);
      }

      return add;
    }(),
    // 缩小
    smaller: (_, __, {
      zoom,
      setZoom
    }) => {
      if (zoom <= 0.5) return;
      setZoom(zoom - 0.1);
    },
    // 放大
    bigger: (_, __, {
      zoom,
      setZoom
    }) => {
      if (zoom >= 1.5) return;
      setZoom(zoom + 0.1);
    },
    // 复制节点
    copy: (_, graph) => {
      const cellItems = graph.getSelectedCells();
      if (!cellItems.length) return message.info('请先选择要复制的节点');
      if (cellItems.length > 1) return message.info('一次只能复制一个节点');
      graph.copy(cellItems);
      message.success('节点成功放入剪贴板');
    },
    // 粘贴节点
    paste: (_, graph, {
      setVisible,
      setCurrentCell
    }) => {
      if (!graph.isClipboardEmpty()) {
        const _graph$getCellsInClip = graph.getCellsInClipboard(),
              _graph$getCellsInClip2 = _slicedToArray(_graph$getCellsInClip, 1),
              cell = _graph$getCellsInClip2[0];

        graph.cleanSelection();
        cell.setData({
          id: null
        }, {
          deep: true
        });
        setCurrentCell(cell);
        nextTick(() => setVisible(true));
      } else {
        message.info('剪贴板中没有节点');
      }
    },
    // 导出图片
    exportImg: (_, graph) => {
      let chartName = undefined;
      Modal.confirm({
        title: '导出图片',
        className: 'rasir-fc-model',
        cancelText: '取消',
        okText: '确定',
        width: 500,
        onOk: () => new Promise((resolve, reject) => {
          if (!chartName) {
            message.info('请输入图片名称');
            reject();
          } else {
            graph.toPNG(dataUri => {
              // 下载
              DataUri.downloadDataUri(dataUri, `${chartName}.png`);
              resolve(1);
            }, {
              quality: 1,
              padding: {
                top: 20,
                right: 20,
                bottom: 20,
                left: 20
              }
            });
          }
        }),
        content: /*#__PURE__*/React.createElement(Input, {
          value: chartName,
          onChange: e => chartName = e.target.value,
          placeholder: "\u8BF7\u8F93\u5165\u56FE\u7247\u540D\u79F0"
        })
      });
    },
    // 保存到后端
    save: (_, graph) => {
      if (!onSave) return;
      Modal.confirm({
        title: '数据上传',
        className: 'rasir-fc-model',
        content: `该操作将数据上传到云端，请确认`,
        okText: '确定',
        cancelText: '取消',
        centered: true,
        onOk: () => new Promise( /*#__PURE__*/function () {
          var _ref5 = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee5(resolve, reject) {
            var ret;
            return regeneratorRuntime.wrap(function _callee5$(_context5) {
              while (1) switch (_context5.prev = _context5.next) {
                case 0:
                  _context5.next = 2;
                  return onSave(getGraphJSONData(graph));

                case 2:
                  ret = _context5.sent;

                  if (ret) {
                    message.success('上传成功！');
                    resolve(1);
                  } else {
                    reject();
                  }

                case 4:
                case "end":
                  return _context5.stop();
              }
            }, _callee5);
          }));

          return function (_x8, _x9) {
            return _ref5.apply(this, arguments);
          };
        }())
      });
    },
    // 导出数据文件
    exportData: (_, graph) => {
      let fileName = '';
      Modal.confirm({
        title: '导出文件',
        className: 'rasir-fc-model',
        content: /*#__PURE__*/React.createElement("div", null, /*#__PURE__*/React.createElement(Input, {
          onChange: e => fileName = e.target.value,
          placeholder: "\u8BF7\u8F93\u5165\u6587\u4EF6\u540D\u79F0"
        })),
        okText: '确定',
        cancelText: '取消',
        onOk: () => {
          if (!fileName) {
            message.error('请输入文件名称');
            return Promise.reject();
          }

          try {
            const content = JSON.stringify(getGraphJSONData(graph));
            const file = new Blob([content], {
              type: 'application/json'
            });
            const fileUrl = URL.createObjectURL(file);
            const linkElement = document.createElement('a');
            linkElement.setAttribute('href', fileUrl);
            linkElement.setAttribute('download', `${fileName}.json`);
            linkElement.click();
            return Promise.resolve(1);
          } catch (e) {
            message.error('数据导出失败');
            return Promise.resolve(1);
          }
        }
      });
    },
    // 导入文件数据
    importData: _ => {
      Modal.confirm({
        title: '导入数据',
        okText: '确定',
        cancelText: '取消',
        className: 'rasir-fc-model',
        content: /*#__PURE__*/React.createElement(Dragger, {
          action: "/",
          fileList: [],
          beforeUpload: file => {
            Modal.destroyAll();
            fileExplain(file);
            return Promise.reject();
          }
        }, /*#__PURE__*/React.createElement("p", {
          style: {
            fontSize: 40
          }
        }, "+"), /*#__PURE__*/React.createElement("p", null, "\u8BF7\u9009\u62E9\u8981\u5BFC\u5165\u7684\u6570\u636E"))
      });
    },
    // 清理缓存数据
    clearStorage: _ => {
      Modal.confirm({
        title: '清理缓存',
        className: 'rasir-fc-model',
        content: `该操作将清理缓存数据，请确认`,
        okText: '确定',
        cancelText: '取消',
        centered: true,
        onOk: function () {
          var _onOk = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee6() {
            return regeneratorRuntime.wrap(function _callee6$(_context6) {
              while (1) switch (_context6.prev = _context6.next) {
                case 0:
                  _context6.next = 2;
                  return autoSaveJson(null, null);

                case 2:
                  window.location.reload();

                case 3:
                case "end":
                  return _context6.stop();
              }
            }, _callee6);
          }));

          function onOk() {
            return _onOk.apply(this, arguments);
          }

          return onOk;
        }()
      });
    },
    // 全屏
    full: (_, __, {
      fullContainer,
      setIsFullScreen
    }) => {
      launchFullScreen(fullContainer);
      nextTick(() => setIsFullScreen(true));
    },
    // 取消全屏
    normal: (_, graph, {
      setIsFullScreen
    }) => {
      exitFullscreen();
      nextTick(() => setIsFullScreen(false));
    },
    // 撤销
    undo: (_, graph) => {
      const history = graph.history;
      history.undo();
    },
    // 重做
    redo: (_, graph) => {
      const history = graph.history;
      history.redo();
    }
  };
  return {
    actionFunMap
  };
}